from tkinter.font import names

from mysqldbHelper import DBHelper

db = DBHelper()

#显示菜单
def show_menu():

    while True:
        print("-----欢迎来学生信息管理系统-----")
        print("---------1. 添加学生---------")
        print("---------2. 查询学生---------")
        print("---------3. 删除学生---------")
        print("---------4. 修改学生---------")
        print("---------5. 查询所有---------")
        print("---------6. 退出系统---------")
        print("-------------------------------")

        # 获取用户输入
        choice = int(input("请输入你的选择："))

        if choice == 1:
            add_student()
        elif choice == 2:
            select_student()
        elif choice == 3:
            delete_student()
        elif choice == 4:
            update_student()
        elif choice == 5:
            get_all()
        elif choice == 6:
            print("欢迎下次再来！")
            break
        else:
            print("输入错误，请重新输入！")
            show_menu()

#增加学生
def add_student():
    name = input("请输入学生信息：")
    age = int(input("请输入学生年龄："))
    sex = input("请输入学生性别：")

    #编写sql命令
    sql = "insert into tb_student (stu_name,stu_age, stu_sex) values(%s, %s, %s)"
    db.update(sql,(name, age, sex))
    print("添加成功！")

#查询学生
def select_student():
    name = input('请输入查询学生的名字：')
    sql = "select * from  tb_student where stu_name = %s"

    #调用执行查询的方法
    result = db.select(sql,name)
    #显示查询的结果
    print('----------查询结果如下------------')



    for row in result:
        print('{:<16}{:<16}{:<16}{:<16}'.format(row[0],row[1],row[2],row[3]))


#删除学生
def delete_student():
    name = int(input('请输入要被删除的id名字：'))
    sql = "delete from tb_student where id = %s"
    db.delete(sql,name)
    print('删除成功')

#查询所有
def get_all():
    sql = "select * from  tb_student"

    #调用执行查询的方法
    result = db.select(sql)
    #显示查询的结果
    print('----------查询结果如下------------')

    for row in result:
        print('{:<16}{:<16}{:<16}{:<16}'.format(row[0],row[1],row[2],row[3]))

#修改学生信息
def update_student():
    id = input('请输入要修改学生的编号：')
    name = input('请输入要修改学生的名字：')
    age = input('请输入要修改学生的年龄：')
    sex = input('请输入要修改学生的性别：')

    sql = "update tb_student set stu_name = %s,stu_age = %s ,stu_sex = %s where id = %s"
    db.update(sql,(name,age,sex,id))
    print("修改完成")



if  __name__ == "__main__":
    show_menu()